################### transformer-units
# transformer-sbt-train:
CUDA_VISIBLE_DEVICES=6 python -m tasks.ClassSummary.models.mTransformer.main \
  --save_path ./model/transformer-sbt/ \
  --input_flag sbt \
  --train ./tasks/ClassSummary/resource/train.token.jsonl \
  --val ./tasks/ClassSummary/resource/valid.token.jsonl \
  --code_field ./tasks/ClassSummary/resource/sbt.field \
  --nl_field ./tasks/ClassSummary/resource/nl.field \
  --patience 4 \
  --batch_size 32 \
  --max_epoch 100 \
  --embedding 1024 \
  --dffn 2048 \
  --layers 3 \
  --nhead 4 \
  --lr 0.0001 \
  --max_grad 5 \
  --gpu_num 1 \
  --seed 1024 \
  --check_freq 1

# transformer-units-hybrid-test:
CUDA_VISIBLE_DEVICES=7 python -m tasks.ClassSummary.models.mTransformer.test \
  --save_path ./model/transformer-units-hybrid.jsonl \
  --input_flag sbt \
  --test ./tasks/ClassSummary/resource/test.token.jsonl \
  --code_field ./tasks/ClassSummary/resource/code.field \
  --nl_field ./tasks/ClassSummary/resource/sbt.field \
  --check_point /home/hlcs/lmc/clsum/model/transformer-code/Transformer-sbt-epoch=06-bleu=1.38.ckpt \
  --save_path ./translate/transformer-sbt-hybird.jsonl \
  --gpu_num 1 \
  --batch_size 64 \
  --seed 1024

# transformer--units-cross-test:
CUDA_VISIBLE_DEVICES=7 python -m tasks.ClassSummary.models.mTransformer.test \
  --save_path ./model/transformer-units-cross.jsonl \
  --input_flag sbt \
  --test ./tasks/ClassSummary/resource/cross.token.jsonl \
  --code_field ./tasks/ClassSummary/resource/sbt.field \
  --nl_field ./tasks/ClassSummary/resource/nl.field \
  --check_point /home/hlcs/lmc/clsum/model/transformer-code/Transformer-sbt-epoch=06-bleu=1.38.ckpt \
  --save_path ./translate/transformer-sbt-cross.jsonl \
  --gpu_num 1 \
  --batch_size 64 \
  --seed 1024
